# -*- coding: utf-8 -*-
# 作者： 安智军
# @Time：2022/6/16 22:10
# @Email：1010666711@qq.com
# @File： main.py
from dev_01_day.book import Book, BookMange


def welcome():
    print("*********************** 欢迎进入图书管理系统***************************")
    print('1、显示所有图书\n2、添加图书\n3、删除图书\n4、查找图书\n5、退出\n6、清空数据\n7、保存数据')
    print('********************************************************************')


def get_choose_numbwer():
    '''
    获取用户输入的菜单编号
    :return:
    '''

    choose_number = input('请输入菜单编号：')  # 返回是个字符串
    # 如说编号不是数字或者不是1、2、3、4、5 返回-1
    if not choose_number.isdigit() or choose_number not in ["1", '2', '3', '4', '5','6','7']:  # isdigit()方法判断是不是数字
        return -1
    return int(choose_number)


def main():
    bm = BookMange() # 创建一个用来管理图书的对象
    while True:
        welcome()
        number = get_choose_numbwer()
        if number == -1:
            print("编号输入有误请重新输入")
            continue
        if number == 1:
            bm.show_book()
        elif number == 2:
            book_id = bm.last_book_id()+1
            book_name = input('请输入书名：')
            book_price = input('请输入书价格：')
            book_loc = input('请输入书的位置：')

            # 创建一本书 对象 具体的书
            book = Book(num=book_id,name=book_name,loc=book_loc,price=book_price)

            bm.add_book(book)
        elif number == 3: # 删除图书
            book_name = input('请输入书名：')
            bm.del_book_by_name(name=book_name)
        elif number == 4:
            pass
        elif number == 6: # 清空数据
            bm.clean_data()
        elif number == 7: # 清空数据
            bm.save()
        else:
            # exit(0)
            print('退出成功')
            break


if __name__ == '__main__':
    main()
